package com.atguigu.flyweight;

import java.util.HashMap;

/**
 * @program: design-patterns
 * @author: yangshen
 * @description: 网站工厂类，根据需要返回一个网站（聚合WebSit，创建一个WebSit的池，用来取WebSit(ConcreteWebSit) ）
 * @create: 2022-02-23 15:58
 **/
public class WebSitFactory {
    /**
     * 集合充当池的作用
     */
    private HashMap<String, ConcreteWebSit> pool = new HashMap<String, ConcreteWebSit>();

    /**
     * 根据网站的类型返回一个网站，如果没有就创建一个网站，并放到池中
     * @return 返回一个网站
     */
    public WebSit getWebSitCategory(String type){
        if (!pool.containsKey(type)){
            //如果没有就创建一个网站，并放到池中
            pool.put(type, new ConcreteWebSit(type));
        }
        return pool.get(type);
    }

    /**
     * 获取网站分类的总数（池中有多少个网站类型）
     * @return 池中有多少个网站类型
     */
    public int getWebSitCount(){
        return pool.size();
    }
}
